tensor([[0.6400, 0.6400, 0.6400, 0.6400, 0.6400],
        [0.3600, 0.3600, 0.3600, 0.3600, 0.3600]], grad_fn=<RoundBackward1>)